// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <stdint.h>

#include "base/macros.h"
#include "media/cast/net/cast_transport_defines.h"
#include "media/cast/net/rtp/frame_buffer.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace media {
namespace cast {

    class FrameBufferTest : public ::testing::Test {
    protected:
        FrameBufferTest()
        {
            payload_.assign(kMaxIpPacketSize, 0);
            rtp_header_.frame_id = FrameId::first();
            rtp_header_.reference_frame_id = FrameId::first();
        }

        ~FrameBufferTest() override { }

        FrameBuffer buffer_;
        std::vector<uint8_t> payload_;
        RtpCastHeader rtp_header_;

        DISALLOW_COPY_AND_ASSIGN(FrameBufferTest);
    };

    TEST_F(FrameBufferTest, OnePacketInsertSanity)
    {
        rtp_header_.rtp_timestamp = RtpTimeTicks().Expand(UINT32_C(3000));
        rtp_header_.is_key_frame = true;
        rtp_header_.frame_id = FrameId::first() + 5;
        rtp_header_.reference_frame_id = FrameId::first() + 5;
        buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_);
        EncodedFrame frame;
        EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame));
        EXPECT_EQ(EncodedFrame::KEY, frame.dependency);
        EXPECT_EQ(FrameId::first() + 5, frame.frame_id);
        EXPECT_EQ(FrameId::first() + 5, frame.referenced_frame_id);
        EXPECT_EQ(3000u, frame.rtp_timestamp.lower_32_bits());
    }

    TEST_F(FrameBufferTest, EmptyBuffer)
    {
        EXPECT_FALSE(buffer_.Complete());
        EncodedFrame frame;
        EXPECT_FALSE(buffer_.AssembleEncodedFrame(&frame));
    }

    TEST_F(FrameBufferTest, DefaultOnePacketFrame)
    {
        buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_);
        EXPECT_TRUE(buffer_.Complete());
        EXPECT_FALSE(buffer_.is_key_frame());
        EncodedFrame frame;
        EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame));
        EXPECT_EQ(payload_.size(), frame.data.size());
    }

    TEST_F(FrameBufferTest, MultiplePacketFrame)
    {
        rtp_header_.is_key_frame = true;
        rtp_header_.max_packet_id = 2;
        buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_);
        ++rtp_header_.packet_id;
        buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_);
        ++rtp_header_.packet_id;
        buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_);
        ++rtp_header_.packet_id;
        EXPECT_TRUE(buffer_.Complete());
        EXPECT_TRUE(buffer_.is_key_frame());
        EncodedFrame frame;
        EXPECT_TRUE(buffer_.AssembleEncodedFrame(&frame));
        EXPECT_EQ(3 * payload_.size(), frame.data.size());
    }

    TEST_F(FrameBufferTest, IncompleteFrame)
    {
        rtp_header_.max_packet_id = 4;
        buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_);
        ++rtp_header_.packet_id;
        buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_);
        ++rtp_header_.packet_id;
        // Increment again - skip packet #2.
        ++rtp_header_.packet_id;
        buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_);
        ++rtp_header_.packet_id;
        buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_);
        EXPECT_FALSE(buffer_.Complete());
        // Insert missing packet.
        rtp_header_.packet_id = 2;
        buffer_.InsertPacket(&payload_[0], payload_.size(), rtp_header_);
        EXPECT_TRUE(buffer_.Complete());
    }

} // namespace media
} // namespace cast
